package pers.ijfxu.testngext.dataprovider;

import java.util.HashMap;
import java.util.List;
import java.util.Map;

import pers.ijfxu.testngext.utils.CollectUtils;

/**
 * 测试数据集合类
 *
 * @author ijfxu  
 * @since 2021/1/27 14:06
 */
public class TestDataCollection {
    private static Map<String, List<String>> datas = new HashMap<>();

    /**
     * 测试数据注册方法，将数据文件和测试方法绑定
     *
     * @param methodName 测试方法名
     * @param filePaths  测试方法对应的测试数据文件
     * @author ijfxu 
     * @since 2021/1/27 14:06
     */
    public static void register(String methodName, String... filePaths) {
        datas.put(methodName, CollectUtils.newArrayList(filePaths));
    }

    /**
     * 根据方法名获取绑定的测试数据文件
     *
     * @param methodName 测试方法名
     * @return java.util.List
     * @author ijfxu 
     * @since 2021/1/27 14:06
     */
    public static List<String> getData(String methodName) {
        return datas.get(methodName);
    }
}
